package cn.sccdlg.fzmybatis.mybatis2sx.utils;

import cn.sccdlg.fzmybatis.mybatis2sx.cfg.Mapper;

import java.beans.PropertyDescriptor;
import java.lang.reflect.Method;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.util.ArrayList;
import java.util.List;

/**
 * @author RaoShuang
 * @emil 991207823@qq.com
 * @date 2020/10/11 22:25
 * mybatis自实现之负责执行SQL语句，并且封装结果集
 */
public class Executor {

    /**
     * 执行sql语句并封装结果集返回实体对象或实体列表
     * @param mapper 关系映射实例
     * @param conn mysql访问连接实例
     * @param <E> 实例类名
     */
    public <E> List<E> selectList(Mapper mapper, Connection conn) {
        // 定义执行平台
        PreparedStatement pstm = null;
        // 定义结果结对象
        ResultSet rs = null;
        try {
            //1.取出mapper中的数据
            //1-1、获得sql语句
            String queryString = mapper.getQueryString();
            //1-2、获取封装映射实体的全限定类名
            String resultType = mapper.getResultType();
            //1-3、获得映射实体对应类对象
            Class domainClass = Class.forName(resultType);
            //2.获取PreparedStatement对象
            pstm = conn.prepareStatement(queryString);
            //3.执行SQL语句，获取结果集
            rs = pstm.executeQuery();
            //4.封装结果集
            //4-1、定义结果列表容器，用于存储查询得到实例
            List<E> list = new ArrayList<>();
            while(rs.next()) {
                //4-2、实例化要封装的实体类对象
                E obj = (E)domainClass.newInstance();

                //4-3、取出结果集的元信息：ResultSetMetaData
                ResultSetMetaData rsmd = rs.getMetaData();
                //4-4、取出总列数
                int columnCount = rsmd.getColumnCount();
                //4-5、遍历总列数
                for (int i = 1; i <= columnCount; i++) {
                    //4-6、获取每列的名称，列名的序号是从1开始的
                    String columnName = rsmd.getColumnName(i);
                    //4-7、根据得到列名，获取每列的值
                    Object columnValue = rs.getObject(columnName);
                    //4-8、给obj赋值：使用Java内省机制（借助PropertyDescriptor实现属性的封装）
                    //      要求：实体类的属性和数据库表的列名保持一种
                    PropertyDescriptor pd = new PropertyDescriptor(columnName,domainClass);
                    //4-9、获取它的写入方法
                    Method writeMethod = pd.getWriteMethod();
                    //4-10、把获取的列的值，给对象赋值
                    writeMethod.invoke(obj,columnValue);
                }
                //4-11、把赋好值的对象加入到集合中
                list.add(obj);
            }
            return list;
        } catch (Exception e) {
            throw new RuntimeException(e);
        } finally {
            release(pstm,rs);
        }
    }

    /**
     * 释放mysql执行过程占用的资源
     * @param pstm 执行对象
     * @param rs 查询结果集对象
     */
    private void release(PreparedStatement pstm,ResultSet rs){
        if(rs != null){
            try {
                rs.close();
            } catch (Exception e) {
                e.printStackTrace();
            }
        }

        if(pstm != null){
            try {
                pstm.close();
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }
}
